package com.heima.article.service;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.heima.article.entity.ApArticle;
import com.heima.article.entity.ApArticleContent;
import com.heima.article.service.IApArticleContentService;
import com.heima.article.service.IApArticleService;
import com.heima.common.minio.service.MinIOService;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;

/**
 * @author: itheima
 * @create: 2022-05-19 11:23
 */
@Service
public class ArticleHtmlTask {
    /**
     *  以下都是此freemarker.template包下
     */
    @Autowired
    private Configuration configuration; //导入了freemarker依赖so可直接注入

    @Autowired
    private IApArticleService articleService;

    @Autowired
    private IApArticleContentService contentService;

    @Autowired
    private MinIOService minIOService;


    /**
     * 生成文章静态页文件，将文件上传到MinIo，更新文章静态页字段
     */
@Async("leadnewsExecutor")//从当前配置的线程池中得到一个线程去执行
    public void createHtmlUpload(Long articleId) {
        try {
            //1.查询文章信息--文章、文章内容（文章、文章内容服务与模板 给模具填充参数的）

            ApArticle article = articleService.getById(articleId);

            LambdaQueryWrapper<ApArticleContent> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(ApArticleContent::getArticleId, articleId);
            ApArticleContent articleContent = contentService.getOne(queryWrapper);


            //2.调用配置对象  加载模板文件得到模板对象
            Template template = configuration.getTemplate("article.ftl"); //"article.ftl"模板名称

            //3.生成html文件
            //3.1 准备模板需要模型数据
            Map<String, Object> modelMap = new HashMap<>();
            //此处名称和模板的必须一致 <#if content??>
            //                <#list content as item> 必须和list处的一致
            modelMap.put("content", JSON.parseArray(articleContent.getContent()));//需要文章正文非文章内容对象   JSON.parseArray(String xxx,Type[] xxx))  JSON.parseArray(String xxx) String的转为集合
            modelMap.put("item", article);
            //3.2 写数据对象   写到StringWriter对象中即writer中
            StringWriter writer = new StringWriter();
            template.process(modelMap, writer); //把模型数据modelMap和模板template组合在一起调用process方法然后将其写到 StringWriter的对象中
            //4.将文件上传MinIO
            InputStream inputStream = new ByteArrayInputStream(writer.toString().getBytes());//将写入对象writer中包含的html字符串再转为字节数组byte[]  封装在InputStream接收

            String htmlUrl = minIOService.upload(articleId + ".html", inputStream, "text/html"); //此时文件名称不可用UUID生成  根据当前文章id找到当前文章对应的Html文件

            //5.更新文章表同步静态页地址
            article.setStaticUrl(htmlUrl);
            article.setSyncStatus(1);
            articleService.updateById(article);
        } catch (Exception e) {
            e.printStackTrace();
        }

    }
}
